/*
 * @lc app=leetcode.cn id=6 lang=cpp
 * @lcpr version=30204
 *
 * [6] Z 字形变换
 */

// @lcpr-template-start
using namespace std;
#include <algorithm>
#include <array>
#include <bitset>
#include <climits>
#include <deque>
#include <functional>
#include <iostream>
#include <list>
#include <queue>
#include <stack>
#include <tuple>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
// @lcpr-template-end
// @lc code=start
class Solution
{
public:
    string convert(string s, int numRows)
    {
        if (numRows == 1 || numRows > s.size()) // 只有一行 或者 只有一列 可以直接返回字符串
            return s;
        vector<string> rows(numRows);
        int i = 0, flag = -1;
        for (char c : s)
        {
            rows[i].push_back(c);
            if (i == 0 || i == numRows - 1)
                flag = -flag;                   // 利用flag控制字符插入的行号
            i += flag;
        }
        string res;
        for (const string &row : rows)          // 每行没有空余的，可以直接把每行的字符串相加得到答案
            res += row;
        return res;
    }
};
// @lc code=end

/*
// @lcpr case=start
// "PAYPALISHIRING"\n3\n
// @lcpr case=end

// @lcpr case=start
// "PAYPALISHIRING"\n4\n
// @lcpr case=end

// @lcpr case=start
// "A"\n1\n
// @lcpr case=end

 */
